# -*- coding: utf-8 -*-
# @Time    : 2024/9/11 下午8:58
# @Author  : liujun
# @Email   : 2592730854@qq.com
# @File    : tools.py
# @Software: PyCharm

import os
import hashlib
import zipfile

def md5_code(passwd):
    md5= hashlib.md5()
    md5.update(passwd.encode("utf-8"))
    return md5.hexdigest ()
def trans_str(s) :
    s=s.replace('":','=')
    return s
def zip_dir(startdir,zip_file):
    #压缩文件夹
    _zip_file = os.path.join(startdir,zip_file)
    if os.path.exists(_zip_file):
        #先删除
        os.remove(_zip_file)
    z=zipfile.ZipFile(_zip_file,'w',zipfile.ZIP_DEFLATED)
    for dir_path,dir_names,file_names in os.walk(startdir):
        fpath = dir_path.replace(startdir,'')   #不replace的话，就从根目录开始复制
        fpath = fpath and fpath + os.sep or ''
        for filename in file_names:
            z.write(os.path.join(dir_path,filename),fpath + filename)
    z.close()


if __name__ == '__main__':
    from consts import REPORT_PATH
    import os
    start_dir=os.path.join(REPORT_PATH,'yasuo')
    zip_dir(start_dir,"yasuo.zip")

